package com.fang.algorithm.leecode;

/**
 * @author fanglingxiao
 * @version 1.0
 * @description 剑指Offer22：链表中倒数第K个节点
 * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
 * <p>
 * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
 * <p>
 * 示例：
 * 给定一个链表: 1->2->3->4->5, 和 k = 2.
 * 返回链表 4->5.
 * @date 2022/5/22 5:08 下午
 **/
public class L22_GetKthFromEnd {
    public static void main(String[] args) {
        ListNode head = new ListNode(1, new ListNode(2, new ListNode(3, new ListNode(4, new ListNode(5)))));
        ListNode ans = getKthFromEnd(head, 2);
        printNode(ans);
    }

    private static void printNode(ListNode ans) {
        while (ans != null) {
            System.out.print(ans.val + " ->");
            ans = ans.next;
        }
    }

    public static ListNode getKthFromEnd(ListNode head, int k) {
        if (head == null || k < 1) {
            return head;
        }
        int size = calcSize(head);
        int ansIsNext = size - k;
        if (ansIsNext <= 0) {
            return head;
        }
        ListNode ans = head;
        int index = 0;
        while (index < ansIsNext) {
            ans = ans.next;
            index++;
        }
        return ans;
    }

    private static int calcSize(ListNode head) {
        int size = 0;
        while (head != null) {
            head = head.next;
        }
        return size;
    }
}
